962D - Merge Equals - CodeForces Solution


data structures implementation *1600

Please click on ads to support us..

C++ Code:

#include<bits/stdc++.h>
#define int long long 
#define rep(i,x,y) for(int i=x;i<=y;i++)
#define pre(i,x,y) for(int i=x;i>=y;i--)
using namespace std;
const int INF=0x3f3f3f3f,mod=1e9+7;

int qpow(int a,int n){
	int num=1;
	while(n){
    	if(n&1)num=num*a;
    	a=a*a;
    	n>>=1;
	}
	return num;
}
int qmul(int a,int b,int mod){
    int num=0;
	while(b){
		if(b&1)num=(num+a)%mod;
        a=a*2%mod;
        b>>=1;
    }
	return num;
}
const int N=1e6+10;

int a[N];
multiset<pair<int,int>>s;
vector<int>ans;
void solve(){
	int n;
	cin>>n;
	rep(i,1,n){
		int x;
		cin>>x;
		s.insert({x,i});
	}
	while(s.size()){
		auto i=s.begin();
		auto j=next(i);
		//cout<<i->first<<" "<<j->first<<endl;
		if(i->first==j->first){
			int sum=i->first*2;
			s.insert({sum,j->second});
			s.erase(i);
			s.erase(j);
		}else{
			a[i->second]=i->first;
			s.erase(i);
		}
	}
	rep(i,1,n){
		if(a[i])ans.push_back(a[i]);
	}
	cout<<ans.size()<<endl;
	for(auto i:ans){
		cout<<i<<" ";
	}
}

void debug(){
	
}
signed main(){ 
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	debug();
	int T=1;
	//scanf("%lld",&T);
	while(T--)solve();
	
    return 0;
}
/*

*/
 	  			  	 				    		  		 					


Comments

Submit
0 Comments
More Questions

1618C - Paint the Array
469A - I Wanna Be the Guy
1294A - Collecting Coins
1227A - Math Problem
349A - Cinema Line
47A - Triangular numbers
1516B - AGAGA XOOORRR
1515A - Phoenix and Gold
1515B - Phoenix and Puzzle
155A - I_love_username
49A - Sleuth
1541A - Pretty Permutations
1632C - Strange Test
673A - Bear and Game
276A - Lunch Rush
1205A - Almost Equal
1020B - Badge
1353A - Most Unstable Array
770A - New Password
1646B - Quality vs Quantity
80A - Panoramix's Prediction
1354B - Ternary String
122B - Lucky Substring
266B - Queue at the School
1490A - Dense Array
1650B - DIV + MOD
1549B - Gregor and the Pawn Game
553A - Kyoya and Colored Balls
1364A - XXXXX
1499B - Binary Removals